from langchain.prompts import PromptTemplate

prompt = PromptTemplate(
    input_variables=["query","file_names"],
    template="""作为一名财务检索专家，你有如下的数据库{file_names}可供你查询。现在传递给你一个检索信息{query}，找到你认为的最具有相关性的那条文本内容。

    输入： '帮我查一下2020年中公教育的营业收入是多少'    输出： './alltxt/2020-03-10__中公教育科技股份有限公司__002607__中公教育__2020年__年度报告.txt'
    输入： '找一下三峡水利2008年的上交税款'    输出： './alltxt/2020-03-10__深圳市中金岭南有色金属股份有限公司__000060__中金岭南__2008年__年度报告.txt'       
    你要只输出内容，回复中不要带有输入的内容。
    如果找不到对应的内容，则回答 '未找到对应文本'
             """,
)

# from 第4章_Langchian知识图谱 import llm_chatglm
#
# #这是一个用于生成的剧情梗概的LMchain
# llm = llm_chatglm.ChatGLM()
#
# from langchain.chains import LLM
# chain = LLMChain(llm=llm, prompt=prompt)
# new_query = (chain.run({"query":query,"file_names":file_names}))
# print(new_query)